// 爹找儿子的方法：

// 参数
// 逻辑
// 返回值

// arr 原始数据
// id 想要找哪个节点就把哪个节点的id传进去
const findSon = (arr, id = '') => {
  const reaultArr = []
  // 逻辑部门
  arr.forEach(item => {
    // 当前输入的id 和数组中的，某一个pid一致，说明当前item是这个ID的儿子
    if (item.pid === id) {
      // 找到所有的子集放到了结果中
      reaultArr.push(item)
      // findSon方法就是找儿子的方法
      // 返回的就是儿子的集合 =》 children
      const children = findSon(arr, item.id)
      // 如果children有长度，说明有儿子
      if (children.length) {
        // 给当前item添加一个children属性，再将返回的儿子集合赋给item.children
        item.children = children
      }
    }
  })
  return reaultArr
}

export default findSon
